home *** CD-ROM | disk | FTP | other *** search
/ ftp.mactech.com 2010 / ftp.mactech.com.tar / ftp.mactech.com / machack / Hacks96 / BetterADsecurity.sit / Better AD security / Source / UninstallCode.c < prev    next >
Text File  |  1996-06-21  |  8KB  |  395 lines
  1. /*    NAME:
  2.         UninstallCode.c
  3.  
  4.     WRITTEN BY:
  5.         Dair Grant
  6.                 
  7.     DESCRIPTION:
  8.         Routines for uninstalling various code resources.
  9.  
  10.     ___________________________________________________________________________
  11. */
  12. //=============================================================================
  13. //        Include files
  14. //-----------------------------------------------------------------------------
  15. #include <Gestalt.h>
  16. #include <Timer.h>
  17. #include "ES.h"
  18. #include "ES Address Table.h"
  19. #include "ExtensionShell.h"
  20. #include "UninstallCode.h"
  21.  
  22.  
  23.  
  24.  
  25.  
  26.  
  27.  
  28.  
  29.  
  30.  
  31.  
  32.  
  33.  
  34.  
  35.  
  36. //=============================================================================
  37. //        Private function prototypes
  38. //-----------------------------------------------------------------------------
  39. void    UninstallTrapPatch(short i);
  40. void    UninstallGestaltSelector(short i);
  41. void    UninstallShutdownTask(short i);
  42. void    UninstallVBLTask(short i);
  43. void    UninstallLowMemFilter(short i);
  44. void    UninstallCodeBlock(short i);
  45. void    UninstallTimeManagerTask(short i);
  46.  
  47.  
  48.  
  49.  
  50.  
  51.  
  52.  
  53.  
  54.  
  55.  
  56.  
  57.  
  58.  
  59.  
  60.  
  61. //=============================================================================
  62. //        Global variables
  63. //-----------------------------------------------------------------------------
  64. extern ESParamBlock        gTheParamBlock;
  65. extern ESAddressTable    *gTheESAddressTable;
  66.  
  67.  
  68.  
  69.  
  70.  
  71.  
  72.  
  73.  
  74.  
  75.  
  76.  
  77.  
  78.  
  79.  
  80.  
  81. //=============================================================================
  82. //        UninstallCode : Try and uninstall a code resource from gTheParamBlock.
  83. //-----------------------------------------------------------------------------
  84. void UninstallCode(short i)
  85. {
  86.  
  87.  
  88.     // Case out on the type of the code, and call the uninstall routine
  89.     switch(gTheParamBlock.theCodeResources[i].codeType) {
  90.         case kTrapPatchType:
  91.              UninstallTrapPatch(i);
  92.              break;
  93.              
  94.         case kGestaltSelectorType:
  95.              UninstallGestaltSelector(i);
  96.              break;
  97.              
  98.         case kShutdownTaskType:
  99.              UninstallShutdownTask(i);
  100.              break;
  101.              
  102.         case kVBLTaskType:
  103.              UninstallVBLTask(i);
  104.              break;
  105.         
  106.         case kLowMemFilterType:
  107.              UninstallLowMemFilter(i);
  108.              break;
  109.         
  110.         case kCodeBlockType:
  111.              UninstallCodeBlock(i);
  112.              break;
  113.         
  114.         case kTimeManagerTaskType:
  115.              UninstallTimeManagerTask(i);
  116.              break;
  117.              
  118.         default:
  119.              ;
  120.     }
  121. } 
  122.  
  123.  
  124.  
  125.  
  126.  
  127.  
  128.  
  129.  
  130.  
  131.  
  132.  
  133.  
  134.  
  135.  
  136.  
  137. //=============================================================================
  138. //        UninstallTrapPatch : Uninstall a trap patch.                                                         
  139. //-----------------------------------------------------------------------------
  140. //        Note :    We check the value currently held in the dispatch table. If
  141. //                it's still the same as our address, then it's safe to remove
  142. //                our routine. We do this just by replacing the dispatch table
  143. //                entry with the old value we originally replaced.
  144. //-----------------------------------------------------------------------------
  145. void UninstallTrapPatch(short i)
  146. {    ProcPtr        currentCode, oldCodeAddress, currentDispatch;
  147.     short        trapWord;
  148.     TrapType    trapType;
  149.     Handle        theHnd;
  150.     
  151.  
  152.     
  153.  
  154.      // Find out what we need to know
  155.      trapWord        = gTheParamBlock.theCodeResources[i].theCodeThing.theTrapPatch.trapWord;
  156.      currentCode        = (ProcPtr) gTheParamBlock.theCodeResources[i].theAddress;
  157.      oldCodeAddress    = gTheESAddressTable->theTable[i];
  158.     trapType        = (trapWord & 0x0800) ? ToolTrap : OSTrap;
  159.     
  160.     
  161.     
  162.     // Compare the value in the dispatch table with our own code. If they match,
  163.     // then it's safe to remove the patch. Otherwise we're stuck.
  164.     currentDispatch = (ProcPtr) StripAddress((Ptr) NGetTrapAddress(trapWord, trapType));
  165.     if (currentDispatch == currentCode)
  166.         {
  167.         // Overwrite the dispatch table entry with the old address.
  168.         NSetTrapAddress((UniversalProcPtr) StripAddress(oldCodeAddress), trapWord, trapType);
  169.    
  170.    
  171.         // Recover a handle to the code we installed, and throw it away
  172.         theHnd = RecoverHandle((Ptr) currentCode);
  173.         DisposeHandle(theHnd);
  174.         }
  175. }
  176.  
  177.  
  178.  
  179.  
  180.  
  181.  
  182.  
  183.  
  184.  
  185.  
  186.  
  187.  
  188.  
  189.  
  190.  
  191. //=============================================================================
  192. //        UninstallGestaltSelector : Uninstall a Gestalt Selector.                                                         
  193. //-----------------------------------------------------------------------------
  194. //        Note :    We can only remove Gestalt Selectors if we've got System 7.5.
  195. //
  196. //                Depending on your Gestalt.h, you may need to comment out
  197. //                the local definition of DeleteGestaltValue.
  198. //-----------------------------------------------------------------------------
  199. extern pascal OSErr DeleteGestaltValue(OSType selector)
  200.  THREEWORDINLINE(0x303C, 0x0203, 0xABF1);
  201. void UninstallGestaltSelector(short i)
  202. {    OSErr                    theErr;
  203.     AGestaltSelector        *theGestaltInfo;
  204.     Handle                    theHnd;
  205.     
  206.  
  207.  
  208.     if (gTheParamBlock.systemVersion >= 0x0750)
  209.         {
  210.         // Remove the selector
  211.         theGestaltInfo    = &gTheParamBlock.theCodeResources[i].theCodeThing.theGestaltSelector;
  212.         theErr            = DeleteGestaltValue(theGestaltInfo->theSelector);
  213.  
  214.  
  215.         // Recover a handle to the code we installed, and throw it away
  216.         theHnd = RecoverHandle(gTheParamBlock.theCodeResources[i].theAddress);
  217.         DisposeHandle(theHnd);
  218.         }
  219. }
  220.  
  221.  
  222.  
  223.  
  224.  
  225.  
  226.  
  227.  
  228.  
  229.  
  230.  
  231.  
  232.  
  233.  
  234.  
  235. //=============================================================================
  236. //        UninstallShutdownTask : Uninstall a Shutdown Task.
  237. //-----------------------------------------------------------------------------
  238. void UninstallShutdownTask(short i)
  239. {    Handle        theHnd;
  240.  
  241.  
  242.     // Remove the shutdown task
  243.     ShutDwnRemove(NewShutDwnProc(gTheParamBlock.theCodeResources[i].theAddress));
  244.  
  245.      
  246.      // Recover a handle to the code and dispose of it
  247.      theHnd = RecoverHandle(gTheParamBlock.theCodeResources[i].theAddress);
  248.     DisposeHandle(theHnd);
  249. }
  250.  
  251.  
  252.  
  253.  
  254.  
  255.  
  256.  
  257.  
  258.  
  259.  
  260.  
  261.  
  262.  
  263.  
  264.  
  265. //=============================================================================
  266. //        UninstallVBLTask : Uninstall a VBL task.
  267. //-----------------------------------------------------------------------------
  268. void UninstallVBLTask(short i)
  269. {    Handle        theHnd;
  270.  
  271.  
  272.     // Remove the VBL task
  273.     VRemove((QElemPtr) gTheParamBlock.theCodeResources[i].theAddress);
  274.  
  275.      
  276.      // Recover a handle to the code and dispose of it
  277.      theHnd = RecoverHandle(gTheParamBlock.theCodeResources[i].theAddress);
  278.     DisposeHandle(theHnd);
  279. }
  280.  
  281.  
  282.  
  283.  
  284.  
  285.  
  286.  
  287.  
  288.  
  289.  
  290.  
  291.  
  292.  
  293.  
  294.  
  295. //=============================================================================
  296. //        UninstallLowMemFilter : Remove a low-memory filter.
  297. //-----------------------------------------------------------------------------
  298. void UninstallLowMemFilter(short i)
  299. {    Handle        theHnd;
  300.     ProcPtr        theEntryPoint, currentHead, oldHead, ourCode;
  301.  
  302.  
  303.     
  304.     
  305.     // Get the address of the filter chain
  306.     theEntryPoint = (ProcPtr) gTheParamBlock.theCodeResources[i].theCodeThing.theLowMemFilter.theEntryPoint;
  307.     
  308.     
  309.     
  310.     // Find out where everything else is
  311.     currentHead    = (ProcPtr) *((ProcPtr *) theEntryPoint);
  312.     oldHead        = gTheESAddressTable->theTable[i];
  313.     ourCode        = (ProcPtr) gTheParamBlock.theCodeResources[i].theAddress;
  314.     
  315.     
  316.     
  317.     // Compare the value at the head of the chain with our own code. If
  318.     // they match, then we can remove our filter. Otherwise we're stuck.
  319.     if (currentHead == ourCode)
  320.         {
  321.         // Overwrite the entry at the head of the chain with the original
  322.         *((ProcPtr *) theEntryPoint) = oldHead;
  323.         
  324.         
  325.         
  326.         // Recover a handle to our code, and throw it away
  327.         theHnd = RecoverHandle((Ptr) ourCode);
  328.         DisposeHandle(theHnd);
  329.         }
  330. }
  331.  
  332.  
  333.  
  334.  
  335.  
  336.  
  337.  
  338.  
  339.  
  340.  
  341.  
  342.  
  343.  
  344.  
  345.  
  346. //=============================================================================
  347. //        UninstallCodeBlock : Remove a block of code.
  348. //-----------------------------------------------------------------------------
  349. void UninstallCodeBlock(short i)
  350. {    ProcPtr        theCode;
  351.     Handle        theHnd;
  352.     
  353.  
  354.     
  355.  
  356.      // Get the pointer to the code block
  357.      theCode = (ProcPtr) gTheParamBlock.theCodeResources[i].theAddress;
  358.  
  359.      
  360.      // Recover a handle to the code and dispose of it
  361.      theHnd = RecoverHandle((Ptr) theCode);
  362.     DisposeHandle(theHnd);
  363. }
  364.  
  365.  
  366.  
  367.  
  368.  
  369.  
  370.  
  371.  
  372.  
  373.  
  374.  
  375.  
  376.  
  377.  
  378.  
  379. //=============================================================================
  380. //        UninstallTimeManagerTask : Remove a Time Manager task.
  381. //-----------------------------------------------------------------------------
  382. void UninstallTimeManagerTask(short i)
  383. {    Handle        theHnd;
  384.  
  385.  
  386.  
  387.     // Remove the Time Manager task
  388.     RmvTime((QElemPtr) gTheParamBlock.theCodeResources[i].theAddress);
  389.  
  390.      
  391.      // Recover a handle to the code and dispose of it
  392.      theHnd = RecoverHandle(gTheParamBlock.theCodeResources[i].theAddress);
  393.     DisposeHandle(theHnd);
  394. }
  395.